LESSER: Post-Training Data Selection with Output-Layer Gradients
LESSER:基于输出层梯度的后训练数据选择
🧭 训练数据选择 · 输出层梯度近似
University of Pennsylvania · University of Texas at Austin
用输出层梯度近似全参数梯度进行数据选择,降低特征提取FLOP 9.7倍/3倍且保持下游性能
📌 工作参考价值(4 点)
- 在 LLM 微调数据筛选中,只提取输出层梯度作为样本特征,避免全参数反向传播,数据选择成本大幅下降;适用于电商搜索/推荐场景下的 LLM 微调、指令数据清洗。
- 工程实现上只需 hook 模型最后一层,兼容现有 gradient-based 选择方法(如 LESS),可作为 drop-in wrapper 快速迁移,无需改变训练流程。
- 即使单个样本的梯度排序与全梯度存在差异,选出的 batch 梯度仍然对齐,因此批量训练效果稳定;可以在大规模候选池中做粗筛或迭代选择。
- 对 RLHF/DPO 等后训练数据筛选,输出层梯度可替代全梯度做相似度排序,尤其适合计算资源受限的团队进行高质量样本挑选。